from dataset import VerCodeData
from model import VerCodeIdentifier

if __name__ == '__main__':
    ver_code_data = VerCodeData()
    ver_code_identifier = VerCodeIdentifier()

    # ver_code_identifier.fit(ver_code_data, batch_size=512)
    data, label = ver_code_data[0]
    label_pred = ver_code_identifier.predict(data.transpose(2, 0, 1))
    print(label, label_pred)
